package com.rem.photoablum.utils;

import com.rem.photoablum.model.Album;
import com.rem.photoablum.model.User;
import lombok.Data;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.MultipartHttpServletRequest;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.util.List;

/**
 * @author remembrance
 * @version 1.0
 * @date 2022/3/6 21:41
 */
@Component
@Data
public class PhotoUtil {
    @Value("${rem.photo.basePath}")
    private String basePath;
    private static final Logger log = LoggerFactory.getLogger(PhotoUtil.class);

    public String createAlbum(Album album){
        String userPath = album.getPid() + "/" + album.getAlbumName();
        String path = basePath + "/" + userPath;
        String absolutePath = new File(path).getAbsolutePath();
        log.info("新建的相册路径：{}",absolutePath);
        File albumPath = new File(absolutePath);
        File parentFile = albumPath.getParentFile();
        if (!parentFile.exists()){
            parentFile.mkdir();
            if (!albumPath.exists()){
                albumPath.mkdir();
            }else {
                return "error";
            }
        }else {
            if (!albumPath.exists()){
                albumPath.mkdir();
            }else {
                return "error";
            }
        }
        return absolutePath;
    }

    //单文件上传
    public String upload(MultipartFile file, Album album) {    //注意参数
        try {
            if (file.isEmpty()) {
                return "文件为空";
            }
            // 获取文件名
            String fileName = file.getOriginalFilename();
            log.info("上传的文件名为：{}",fileName);//写日志
            // 获取文件的后缀名
            String suffixName = fileName.substring(fileName.lastIndexOf("."));
            log.info("文件的后缀名为：{}", suffixName);//写日志
            // 设置文件存储路径         *************************************************
//            String userPath = user.getUsername();//获取用户的用户名，根据用户名来创建专属文件夹
//            String path = basePath + "/" + userPath +"/" + fileName;
            String userPath = album.getPath();
            String path = userPath + "/" + fileName;
            String absolutePath = new File(path).getAbsolutePath();
            log.info("文件的上传路径：{}",absolutePath);
            File dest = new File(absolutePath);// dist为文件，有多级目录的文件
            // 检测是否存在目录
            if (!dest.getParentFile().exists()) {//因此这里使用.getParentFile()，目的就是取文件前面目录的路径
                dest.getParentFile().mkdirs();// 新建文件夹
            }
            file.transferTo(dest);// 文件写入
            return absolutePath;//将上传的文件的具体路径返回
        } catch (IllegalStateException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }
        return "上传失败";
    }

    //多文件上传
    //逻辑有问题，待优化。。。
    //TODO
    public String handleFileUpload(HttpServletRequest request,User user) {   //注意参数
        List<MultipartFile> files = ((MultipartHttpServletRequest) request).getFiles("file");
        MultipartFile file = null;
        BufferedOutputStream stream = null;
        for (int i = 0; i < files.size(); ++i) {
            file = files.get(i);
            String fileName = file.getOriginalFilename();
            // 设置文件存储路径         *************************************************
            String userPath = user.getUsername();//获取用户的用户名，根据用户名来创建专属文件夹
            String path = basePath + "/" + userPath +"/" + fileName;
            String filePath = new File(path).getAbsolutePath();
            log.info("文件的上传路径：{}",filePath);         // 文件路径
            File dest = new File(filePath);
            // 检测是否存在目录
            if (!dest.exists()) {
                dest.mkdirs();// 新建文件夹
            }
            if (!file.isEmpty()) {
                try {
                    byte[] bytes = file.getBytes();
                    stream = new BufferedOutputStream(new FileOutputStream(
                            new File(filePath + file.getOriginalFilename())));//设置文件路径及名字
                    stream.write(bytes);// 写入
                    stream.close();
                } catch (Exception e) {
                    stream = null;
                    return "第 " + i + " 个文件上传失败 ==> " + e.getMessage();
                }
            } else {
                return "第 " + i + " 个文件上传失败因为文件为空";
            }
        }
        return "上传成功";
    }

    //单文件下载
    public String downloadFile(HttpServletRequest request, HttpServletResponse response,String photoPath) {
        String fileName = photoPath;// 文件名
        if (fileName != null) {
            //设置文件路径
            File file = new File(fileName);
            log.info("下载的文件名：{}",file.getName());
            if (file.exists()) {
                response.setContentType("application/force-download");// 设置强制下载不打开
                response.addHeader("Content-Disposition", "attachment;fileName=" + fileName);// 设置文件名
                byte[] buffer = new byte[1024];
                FileInputStream fis = null;
                BufferedInputStream bis = null;
                try {
                    fis = new FileInputStream(file);
                    bis = new BufferedInputStream(fis);
                    OutputStream os = response.getOutputStream();
                    int i = bis.read(buffer);
                    while (i != -1) {
                        os.write(buffer, 0, i);
                        i = bis.read(buffer);
                    }
                    return "下载成功";
                } catch (Exception e) {
                    e.printStackTrace();
                } finally { // 做关闭操作
                    if (bis != null) {
                        try {
                            bis.close();
                        } catch (IOException e) {
                            e.printStackTrace();
                        }
                    }
                    if (fis != null) {
                        try {
                            fis.close();
                        } catch (IOException e) {
                            e.printStackTrace();
                        }
                    }
                }
            }
        }
        return "下载失败";
    }


    //单文件删除
    public String deletePhoto(String url){
        File file = new File(url);
        if (file.exists()) {
            file.delete();
            return "删除成功";
        }
        return "删除失败";
    }

    //批量删除（删除整个相册）
    public String deleteAlbum(String path){
        File file = new File(path);
        if (file != null){
            log.info("批量删除的文件路径:{}",path);
            File[] files = file.listFiles();
            if (files.length != 0){
                for (File photo : files) {
                    log.info("删除文件:{}",photo.getAbsolutePath());
                    photo.delete();
                }
                file.delete();
            }else {
                file.delete();
            }
            return "删除成功";
        }
        return "删除失败";
    }


}


